How to take matrix input from the user and display the matrix in Python